#include <3ds.h>
#include <citro3d.h>
#include <string.h>
#include "vshader_shbin.h"

#define CLEAR_COLOR 0x68B0D8FF

#define DISPLAY_TRANSFER_FLAGS \
	(GX_TRANSFER_FLIP_VERT(0) | GX_TRANSFER_OUT_TILED(0) | GX_TRANSFER_RAW_COPY(0) | \
	GX_TRANSFER_IN_FORMAT(GX_TRANSFER_FMT_RGBA8) | GX_TRANSFER_OUT_FORMAT(GX_TRANSFER_FMT_RGB8) | \
	GX_TRANSFER_SCALING(GX_TRANSFER_SCALE_NO))

typedef struct { float x, y, z; } vertex;

static const vertex vertex_list[] = {
	{ 200.0f, 200.0f, 0.5f },
	{ 100.0f, 40.0f, 0.5f },
	{ 300.0f, 40.0f, 0.5f },
};

typedef enum {
    Platform_Citra,
    Platform_Panda,
    Platform_Other
} Platform;

typedef enum {
    SystemInfo_Citra = 0x20000,
    SystemInfo_Panda = 0x20001
} SystemInfoType;

typedef enum {
    SystemInfoSub_IsCitra = 0,
    SystemInfoSub_IsPanda = 0,
} SystemInfoSubType;

// Detect the emulator this is running on
Platform getPlatform() {
    s64 out;

    // First, attempt to detect Citra
    Result res = svcGetSystemInfo(&out, SystemInfo_Citra, SystemInfoSub_IsCitra);
    if (R_SUCCEEDED(res) && out == 1) {
        return Platform_Citra;
    }
    
    // Next, attempt to detect Panda3DS
    res = svcGetSystemInfo(&out, SystemInfo_Panda, SystemInfoSub_IsPanda);
    if (R_SUCCEEDED(res) && out == 1) {
        return Platform_Panda;
    }

    // Unknown platform, maybe a console or another emulator
    return Platform_Other;
}

// Print string in emulator terminal
static void emuPrint(const char* str) {
    svcOutputDebugString(str, strlen(str));
}

#define vertex_list_count (sizeof(vertex_list)/sizeof(vertex_list[0]))

static DVLB_s* vshader_dvlb;
static shaderProgram_s program;
static int uLoc_projection;
static C3D_Mtx projection;

static void* vbo_data;

static void sceneInit(void) {
	// Load the vertex shader, create a shader program and bind it
	vshader_dvlb = DVLB_ParseFile((u32*)vshader_shbin, vshader_shbin_size);
	shaderProgramInit(&program);
	shaderProgramSetVsh(&program, &vshader_dvlb->DVLE[0]);
	C3D_BindProgram(&program);

	// Get the location of the uniforms
	uLoc_projection = shaderInstanceGetUniformLocation(program.vertexShader, "projection");

	// Configure attributes for use with the vertex shader
	C3D_AttrInfo* attrInfo = C3D_GetAttrInfo();
	AttrInfo_Init(attrInfo);
	AttrInfo_AddLoader(attrInfo, 0, GPU_FLOAT, 3); // v0=position
	AttrInfo_AddFixed(attrInfo, 1); // v1=color

	// Set the fixed attribute (color) to a colour depending on the emulator

    Platform platform = getPlatform();
    switch (platform) {
        case Platform_Citra:
            emuPrint("Detected Citra\n");
            C3D_FixedAttribSet(1, 1.0, 1.0, 0.0, 1.0);
            break;
        
        case Platform_Panda:
            emuPrint("Detected Panda3DS\n");
            C3D_FixedAttribSet(1, 1.0, 0.0, 0.0, 1.0);
            break;

        default:
            emuPrint("Unknown platform. Probably a real 3DS\n");
            C3D_FixedAttribSet(1, 1.0, 0.5, 0.2, 1.0);
            break;
    }

	// Compute the projection matrix
	Mtx_OrthoTilt(&projection, 0.0, 400.0, 0.0, 240.0, 0.0, 1.0, true);

	// Create the VBO (vertex buffer object)
	vbo_data = linearAlloc(sizeof(vertex_list));
	memcpy(vbo_data, vertex_list, sizeof(vertex_list));

	// Configure buffers
	C3D_BufInfo* bufInfo = C3D_GetBufInfo();
	BufInfo_Init(bufInfo);
	BufInfo_Add(bufInfo, vbo_data, sizeof(vertex), 1, 0x0);

	// Configure the first fragment shading substage to just pass through the vertex color
	// See https://www.opengl.org/sdk/docs/man2/xhtml/glTexEnv.xml for more insight
	C3D_TexEnv* env = C3D_GetTexEnv(0);
	C3D_TexEnvInit(env);
	C3D_TexEnvSrc(env, C3D_Both, GPU_PRIMARY_COLOR, 0, 0);
	C3D_TexEnvFunc(env, C3D_Both, GPU_REPLACE);
}

static void sceneRender(void) {
	// Update the uniforms
	C3D_FVUnifMtx4x4(GPU_VERTEX_SHADER, uLoc_projection, &projection);

	// Draw the VBO
	C3D_DrawArrays(GPU_TRIANGLES, 0, vertex_list_count);
}

static void sceneExit(void) {
	// Free the VBO
	linearFree(vbo_data);

	// Free the shader program
	shaderProgramFree(&program);
	DVLB_Free(vshader_dvlb);
}

int main() {
    emuPrint("Entering main\n");
	// Initialize graphics
	gfxInitDefault();
	C3D_Init(C3D_DEFAULT_CMDBUF_SIZE);

	// Initialize the render target
	C3D_RenderTarget* target = C3D_RenderTargetCreate(240, 400, GPU_RB_RGBA8, GPU_RB_DEPTH24_STENCIL8);
	C3D_RenderTargetSetOutput(target, GFX_TOP, GFX_LEFT, DISPLAY_TRANSFER_FLAGS);

	// Initialize the scene
	sceneInit();

	// Main loop
	while (true)
	{
		// Render the scene
        emuPrint("Entering C3D_FrameBegin");
		C3D_FrameBegin(C3D_FRAME_SYNCDRAW);
            emuPrint("Clearing render target");
			C3D_RenderTargetClear(target, C3D_CLEAR_ALL, CLEAR_COLOR, 0);
            emuPrint("Calling C3D_FrameDrawOn");
			C3D_FrameDrawOn(target);
            emuPrint("Calling sceneRender");
			sceneRender();
            emuPrint("Entering C3D_FrameEnd");
		C3D_FrameEnd(0);
        emuPrint("Exited C3D_FrameEnd");
	}

	// Deinitialize the scene
	sceneExit();

	// Deinitialize graphics
	C3D_Fini();
	gfxExit();
	return 0;
}
